package com.zwq.leetcode.arr.twonumadd;

import java.util.LinkedList;
import java.util.List;

/**
 * 给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。
 *
 * 如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。
 *
 * 您可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 *
 * 示例：
 *
 * 输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
 * 输出：7 -> 0 -> 8
 * 原因：342 + 465 = 807
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/add-two-numbers
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Solution {
    public static void main(String[] args) {
        ListNode l1 = new ListNode();
        l1.val = 2;
        ListNode l2 = new ListNode();
        l2.val = 4;
        ListNode l3 = new ListNode();
        l3.val = 3;
        l1.next = l2;
        l2.next = l3;

        ListNode ll1 = new ListNode();
        ll1.val = 5;
        ListNode ll2 = new ListNode();
        ll2.val = 6;
        ListNode ll3 = new ListNode();
        ll3.val = 4;
        ll1.next = ll2;
        ll2.next = ll3;

        ListNode result = addTwoNumbers(l1,ll1);
    }

    //使用LinkedList
    public static LinkedList addTwoNumbers(LinkedList l1, LinkedList l2) {
        int size = l1.size() > l2.size() ? l1.size() : l2.size();
        LinkedList res =new LinkedList();
        int base = 0;
        for(int i =0;i< size;i++){
            int num1 = (int) l1.size() < i ? 0: (int) l1.get(i);
            int num2 = (int) l2.size() < i ? 0: (int) l2.get(i);
            int sum = num1 + num2 + base;
            if(sum >= 10){
                base = 1;
            }else {
                base = 0;
            }
            res.add(sum % 10);
        }
        if(base == 1) res.add(1);
        return res;
    }

    //使用ListNode
    public static ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode res =new ListNode();
        ListNode root = res;
        ListNode sumNode;
        int base = 0;
        while(l1 != null || l2!=null){
            int num1 = l1 == null ? 0: (int) l1.val;
            int num2 = l2 == null ? 0: (int) l2.val;
            int sum = num1 + num2 + base;
            if(sum >= 10){
                base = 1;
            }else {
                base = 0;
            }
            sumNode = new ListNode();
            sumNode.val = sum % 10;

            res.next = sumNode;
            res = sumNode;

            if(l1!= null) l1 = l1.next;
            if(l2!= null) l2 = l2.next;
        }

        if(base == 1) {
            ListNode baseNode = new ListNode();
            baseNode.val = 1;
            res.next = baseNode;
        }
        return root.next;
    }
}
class ListNode {
    int val;
    ListNode next;
}
